design of shafts based on rigidity